import m
import torch
import torch.nn.functional as F
class hm0(m.EmotionRecognitionModel):
    def forward(self, x):
        x = self.p0(self.conv1(x.view(-1, len(x))))
        x = self.avg_pool(x)
        x = x.squeeze(0)
        x1 = torch.t(x)
        # 全连接层
        x = self.p1(self.fc1(x1))
        x = self.fc2(x)
        # Softmax分类层
        x = torch.mean(x, dim=0)
        x = F.softmax(x, dim=0)
        return x